"""
TPshop的登录接口封装
1、获取验证码  http://localhost/index.php?m=Home&c=User&a=verify
2、登录  http://localhost/index.php?m=Home&c=User&a=do_login
"""

import requests

# 定义要测试的接口类（TPshop的登录接口封装）
class LoginAPI(object):
    # 要在初始化代码中，保存两个接口地址
    def __init__(self):
        self.url_verify = "http://localhost/index.php?m=Home&c=User&a=verify"
        self.url_login = "http://localhost/index.php?m=Home&c=User&a=do_login"

    # 获取验证码的方法：传递的时候，需要使用session
    def get_verify_code(self,session):
        return session.get(self.url_verify)

    # 登录方法的接口：session、username、password、verify_code

    def login(self,session,username,password,verify_code):
        login_data = {
            "username":username,
            "password":password,
            "verify_code":verify_code
        }

        return session.post(self.url_login,json=login_data)
